package com.bird.设计模式.结构型.代理模式.db;

/**
 * @data 2022/3/16 15:46
 * @author: bird
 * @description: 数据源上下文
 */
public class DataSourceContextHolder {

    //保存的是当前线程的数据源(DB)
    private static final ThreadLocal<String> CONTENT_HOLDER = new ThreadLocal<String>();


    //保存当前线程的数据源
    public static void setDBType(String dbType) {
        CONTENT_HOLDER.set(dbType);
    }

    // 获取当前线程的数据源
    public static String getDBType() {
        return (String) CONTENT_HOLDER.get();
    }

    public static void clearDBType() {
        CONTENT_HOLDER.remove();
    }

}
